package com.ylin.coder.utils;

import freemarker.template.Configuration;
import freemarker.template.Template;

import java.io.*;
import java.util.Locale;
import java.util.Map;

/**
 * 创建人：masl
 * @version
 */
public class Freemarker {

	/**
	 * 打印到控制台(测试用)
	 *  @param ftlName
	 */
	public static void print(String ftlName, Map<String, Object> root, String ftlPath) throws Exception {
		Template temp = getTemplate(ftlName, ftlPath); //通过Template可以将模板文件输出到相应的流
		temp.process(root, new PrintWriter(System.out));
	}

	/**
	 * 输出到输出到文件
	 * @param ftlName   ftl文件名
	 * @param root		传入的map
	 * @param outFile	输出后的文件全部路径
	 * @param filePath	输出前的文件上部路径
	 */
	public static void printFile(String ftlName, Map<String, Object> root, String outFile, String filePath, String ftlPath) throws Exception {
		File file = new File(filePath + outFile);
		if (!file.getParentFile().exists()) { // 判断有没有父路径，就是判断文件整个路径是否存在
			file.getParentFile().mkdirs(); // 不存在就全部创建
		}
		Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8"));
		Template template = getTemplate(ftlName, ftlPath);
		template.process(root, out); // 模版输出
		out.flush();
		out.close();
	}

	/**
	 * 通过文件名加载模版
	 * @param ftlName
	 */
	public static Template getTemplate(String ftlName, String ftlPath) throws Exception {
		Configuration cfg = new Configuration(); // 通过Freemaker的Configuration读取相应的ftl
		cfg.setEncoding(Locale.CHINA, "utf-8");
		cfg.setDirectoryForTemplateLoading(new File(Tools.getClassResources() + "/ftl/" + ftlPath)); // 设定去哪里读取相应的ftl模板文件
		Template temp = cfg.getTemplate(ftlName); // 在模板文件目录中找到名称为name的文件
		return temp;
	}
}
